package com.dhcc.bpm.common.docNode;

import java.util.ArrayList;
import java.util.List;

/**
 * @ClassName ForestDocNodeManger
 * @Description 文档节点
 * @Author sjm
 * @Date 2020/2/21 0021 15:16
 * @Version V1.0
 * @Copyright: 2020 东华软件 . All rights reserved.
 **/
public class ForestDocNodeManger  <T extends ZtreeNode> {

    /**
     * 森林的所有节点
     */
    private List<T> list;

    /**
     * 森林的父节点ID
     */
    private List<String> parentIds = new ArrayList<>();

    public ForestDocNodeManger(List<T> items) {
        list = items;
    }

    /**
     * 根据节点ID获取一个节点
     *
     * @param
     * @return 对应的节点对象
     */
    public ZtreeNode getTreeNodeAT(String code) {
        for (ZtreeNode forestDocNode : list) {
            if (code.equals(forestDocNode.getCode() )) {
                return forestDocNode;
            }
        }
        return null;
    }

    /**
     * 增加父节点ID
     *
     * @param parentId
     */
    public void addParentId(String parentId) {
        parentIds.add(parentId);
    }

    /**
     * 获取树的根节点(一个森林对应多颗树)
     *
     * @return 树的根节点集合
     */
    public List<T> getRoot() {
        List<T> roots = new ArrayList<>();
        for (T forestDocNode : list) {
            if ("0".equals(forestDocNode.getParentId()) || parentIds.contains(forestDocNode.getCode())) {
                roots.add(forestDocNode);
            }
        }
        return roots;
    }

}
